package com.haixiaoke.saas.common.utils.bean;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLInputFactory;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.StringWriter;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Bean 工具类
 *
 * @author haixiaoke
 */
@Slf4j
public class BeanUtils extends org.springframework.beans.BeanUtils {
    /**
     * Bean方法名中属性名开始的下标
     */
    private static final int BEAN_METHOD_PROP_INDEX = 3;

    /**
     * 匹配getter方法的正则表达式
     */
    private static final Pattern GET_PATTERN = Pattern.compile("get(\\p{javaUpperCase}\\w*)");

    /**
     * 匹配setter方法的正则表达式
     */
    private static final Pattern SET_PATTERN = Pattern.compile("set(\\p{javaUpperCase}\\w*)");

    /**
     * Bean属性复制工具方法。
     *
     * @param dest 目标对象
     * @param src  源对象
     */
    public static void copyBeanProp(Object dest, Object src) {
        try {
            copyProperties(src, dest);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取对象的setter方法。
     *
     * @param obj 对象
     * @return 对象的setter方法列表
     */
    public static List<Method> getSetterMethods(Object obj) {
        // setter方法列表
        List<Method> setterMethods = new ArrayList<Method>();

        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();

        // 查找setter方法

        for (Method method : methods) {
            Matcher m = SET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 1)) {
                setterMethods.add(method);
            }
        }
        // 返回setter方法列表
        return setterMethods;
    }

    /**
     * 获取对象的getter方法。
     *
     * @param obj 对象
     * @return 对象的getter方法列表
     */

    public static List<Method> getGetterMethods(Object obj) {
        // getter方法列表
        List<Method> getterMethods = new ArrayList<Method>();
        // 获取所有方法
        Method[] methods = obj.getClass().getMethods();
        // 查找getter方法
        for (Method method : methods) {
            Matcher m = GET_PATTERN.matcher(method.getName());
            if (m.matches() && (method.getParameterTypes().length == 0)) {
                getterMethods.add(method);
            }
        }
        // 返回getter方法列表
        return getterMethods;
    }

    /**
     * 检查Bean方法名中的属性名是否相等。<br>
     * 如getName()和setName()属性名一样，getName()和setAge()属性名不一样。
     *
     * @param m1 方法名1
     * @param m2 方法名2
     * @return 属性名一样返回true，否则返回false
     */

    public static boolean isMethodPropEquals(String m1, String m2) {
        return m1.substring(BEAN_METHOD_PROP_INDEX).equals(m2.substring(BEAN_METHOD_PROP_INDEX));
    }

    /**
     * 拷贝Bean对象的值，从source中拷贝到target中。只拷贝target中与source属性相同且target属性为null的值。<br>
     *
     * @param source 源
     * @param target 目标
     * @param flag 忽略的字段
     * @author qixi
     * @date 2024-4-9 17:34:43
     */
    public static void copyPropertiesIfNull(Object source, Object target, String flag) {
        // 获取源对象的所有字段
        Field[] sourceFields = source.getClass().getDeclaredFields();
        // 获取目标对象的所有字段
        Field[] targetFields = target.getClass().getDeclaredFields();
        try {
            // 遍历
            for (Field sourceField : sourceFields) {
                if (!StrUtil.equals(sourceField.getName(), flag)) {
                    for (Field targetField : targetFields) {
                        // 根据字段名称去匹配，有相同的字段名称进行复制值
                        if (sourceField.getName().equals(targetField.getName())) {
                            // 开启权限
                            sourceField.setAccessible(true);
                            targetField.setAccessible(true);
                            // 判断目标对象里面的字段为null时
                            if (ObjectUtil.isEmpty(targetField.get(target))) {
                                // 给目标对象字段赋值
                                targetField.set(target, sourceField.get(source));
                                // 赋值后跳出当前循环
                                break;
                            }
                            // 关闭权限
                            sourceField.setAccessible(false);
                            targetField.setAccessible(false);
                        }
                    }
                }
            }
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
    }

    /**
     * xml字符串转java对象
     * @param xmlString xml字符串
     * @param clazz 要转的java对象
     * @param <T> 泛型
     *
     * @return T
     */
    public static <T> T xmlToBean(String xmlString, Class<T> clazz){
        try {
            XMLInputFactory factory = XMLInputFactory.newInstance();
            InputStream stream = new ByteArrayInputStream(xmlString.getBytes(StandardCharsets.UTF_8));
            XMLEventReader someSource = factory.createXMLEventReader(stream);

            JAXBContext jaxbContext = JAXBContext.newInstance(clazz);
            Unmarshaller jaxbUnmarshaller = jaxbContext.createUnmarshaller();
            JAXBElement<T> userElement = jaxbUnmarshaller.unmarshal(someSource, clazz);
            return userElement.getValue();
        }catch (Exception e){
            log.error("xml转换bean失败：{}",e.getMessage(),e);
            return null;
        }
    }

    /**
     * bean对象转xml字符串
     * @param obj bean对象
     * @param clazz 要转的bean.Class
     *
     * @return object
     */
    public static String beanToXml(Object obj, Class<?> clazz) {
        try {
            JAXBContext context = JAXBContext.newInstance(clazz);
            Marshaller marshaller = context.createMarshaller();
            marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
            marshaller.setProperty(Marshaller.JAXB_ENCODING, "GBK");
            StringWriter writer = new StringWriter();
            marshaller.marshal(obj, writer);
            return writer.toString();
        }catch (Exception e){
            log.error("bean转换xml失败：{}",e.getMessage(),e);
            return null;
        }
    }
}
